
import unittest
from b import main,extension
class CheckTestCase(unittest.TestCase):
	def testexam1(self):
		testcheck1=main('a','file\hello.txt')
		self.assertEqual(testcheck1,13)
	def testexam2(self):
		testcheck2=extension('r','file\hello.txt')
		self.assertEqual(testcheck2,1)
unittest.main()
